#pragma once
#include <cuda_runtime.h> // cuda相关头文件
#include <stdint.h>
#include <stdio.h>
#include <string>
static bool set_gpu_id(const int id = 0)
{
	cudaError_t status = cudaSetDevice(id);
	if (status != cudaSuccess)
	{
		std::cout << "gpu id :" + std::to_string(id) + " not exist !" << std::endl;
		return false;
	}
	return true;
}